# Copyright 2020-2022 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================

"""tbe ops"""
from .celu import _celu_tbe
from .cummin import _cummin_tbe
from .abs import _abs_tbe
from .abs_ds import _abs_ds_tbe
from .inplace_add import _inplace_add_tbe
from .inplace_sub import _inplace_sub_tbe
from .abs_grad import _abs_grad_tbe
from .abs_grad_ds import _abs_grad_ds_tbe
from .acos import _acos_tbe
from .acos_ds import _acos_ds_tbe
from .acos_grad import _acos_grad_tbe
from .acos_grad_ds import _acos_grad_ds_tbe
from .acosh import _acosh_tbe
from .acosh_ds import _acosh_ds_tbe
from .acosh_grad import _acosh_grad_tbe
from .acosh_grad_ds import _acosh_grad_ds_tbe
from .adam_apply_one_with_decay import _adam_apply_one_with_decay_tbe
from .apply_centered_rms_prop import _apply_centered_rms_prop_tbe
from .apply_centered_rms_prop_ds import _apply_centered_rms_prop_ds_tbe
from .add import _add_tbe
from .add_ds import _add_ds_tbe
from .add_n import _add_n_tbe
from .add_n_ds import _add_n_ds_tbe
from .addcdiv import _addcdiv_tbe
from .addcdiv_ds import _addcdiv_ds_tbe
from .addcmul import _addcmul_tbe
from .addcmul_ds import _addcmul_ds_tbe
from .accumulate_n_v2 import _accumulate_n_v2_tbe
from .accumulate_n_v2_ds import _accumulate_n_v2_ds_tbe
from .apply_ftrl import _apply_ftrl_tbe
from .apply_ftrl_ds import _apply_ftrl_ds_tbe
from .apply_keras_momentum import _apply_keras_momentum_tbe
from .apply_momentum import _apply_momentum_tbe
from .apply_momentum_ds import _apply_momentum_ds_tbe
from .apply_adam import _apply_adam_tbe
from .apply_adam_ds import _apply_adam_ds_tbe
from .apply_ada_max import _apply_ada_max_tbe
from .apply_ada_max_ds import _apply_ada_max_ds_tbe
from .apply_adadelta import _apply_adadelta_tbe
from .apply_adadelta_ds import _apply_adadelta_ds_tbe
from .apply_adagrad import _apply_adagrad_tbe
from .apply_adagrad_ds import _apply_adagrad_ds_tbe
from .apply_adagrad_v2 import _apply_adagrad_v2_tbe
from .apply_adagrad_v2_ds import _apply_adagrad_v2_ds_tbe
from .apply_adagrad_d_a import _apply_adagrad_d_a_tbe
from .apply_add_sign import _apply_add_sign_tbe
from .apply_add_sign_ds import _apply_add_sign_ds_tbe
from .apply_power_sign import _apply_power_sign_tbe
from .apply_power_sign_ds import _apply_power_sign_ds_tbe
from .apply_gradient_descent import _apply_gradient_descent_tbe
from .apply_gradient_descent_ds import _apply_gradient_descent_ds_tbe
from .apply_proximal_gradient_descent import _apply_proximal_gradient_descent_tbe
from .apply_proximal_gradient_descent_ds import _apply_proximal_gradient_descent_ds_tbe
from .sparse_apply_ftrl_v2 import _sparse_apply_ftrl_v2_tbe
from .sparse_apply_adagrad_v2 import _sparse_apply_adagrad_v2_tbe
from .approximate_equal import _approximate_equal_tbe
from .approximate_equal_ds import _approximate_equal_ds_tbe
from .adam_apply_one import _adam_apply_one_tbe
from .assign import _assign_tbe
from .assign_ds import _assign_ds_tbe
from .assign_add import _assign_add_tbe
from .assign_add_ds import _assign_add_ds_tbe
from .assign_sub import _assign_sub_tbe
from .assign_sub_ds import _assign_sub_ds_tbe
from .batch_matmul import _batch_matmul_tbe
from .batch_matmul_ds import _batch_matmul_ds_tbe
from .batch_matmul_v2 import _batch_matmul_v2_tbe
from .batchnorm import _batch_norm_tbe
from .batchnorm_grad import _batch_norm_grad_tbe
from .bias_add import _bias_add_tbe
from .bias_add_ds import _bias_add_ds_tbe
from .bias_add_grad import _bias_add_grad_tbe
from .cast import _cast_tbe
from .cast_ds import _cast_ds_tbe
from .conv2d import _conv2d_tbe
from .conv2d_backprop_filter import _conv2d_backprop_filter_tbe
from .conv2d_backprop_input import _conv2d_backprop_input_tbe
from .conv2d_ds import _conv2d_ds_tbe
from .conv2d_backprop_filter_ds import _conv2d_backprop_filter_ds_tbe
from .conv2d_backprop_input_ds import _conv2d_backprop_input_ds_tbe
from .conv2d_transpose import _conv2d_transpose_tbe
from .confusion_mul_grad import _confusion_mul_grad_tbe
from .dilation import _dilation_tbe
from .dropout_do_mask import _dropout_do_mask_tbe
from .dropout_do_mask_ds import _dropout_do_mask_ds_tbe
from .dropout_do_mask_v3 import _dropout_do_mask_v3_tbe
from .softmax_v2_with_dropout_do_mask_v3 import _softmax_v2_with_dropout_do_mask_v3_tbe
from .gelu import _gelu_tbe
from .gelu_ds import _gelu_ds_tbe
from .gelu_grad import _gelu_grad_tbe
from .gelu_grad_ds import _gelu_grad_ds_tbe
from .ger import _ger_tbe
from .ger_ds import _ger_ds_tbe
from .fast_gelu import _fast_gelu_tbe
from .fast_gelu_ds import _fast_gelu_ds_tbe
from .fast_gelu_grad import _fast_gelu_grad_tbe
from .fast_gelu_grad_ds import _fast_gelu_grad_ds_tbe
from .max_pool import _max_pool_tbe
from .max_pool_ds import _max_pool_ds_tbe
from .max_pool_grad import _max_pool_grad_tbe
from .max_pool_grad_with_argmax import _max_pool_grad_with_argmax_tbe
from .max_pool_with_argmax import _max_pool_with_argmax_tbe
from .max_pool3d import _max_pool_3d_tbe
from .max_pool3d_grad import _max_pool_3d_grad_tbe
from .max_pool3d_grad_grad import _max_pool_3d_grad_grad_tbe
from .mul import _mul_tbe
from .mul_ds import _mul_ds_tbe
from .real_div import _real_div_tbe
from .real_div_ds import _real_div_ds_tbe
from .relu import _relu_tbe
from .relu_ds import _relu_ds_tbe
from .relu_grad import _relu_grad_tbe
from .relu_grad_ds import _relu_grad_ds_tbe
from .relu6 import _relu6_tbe
from .relu6_ds import _relu6_ds_tbe
from .relu6_grad import _relu6_grad_tbe
from .relu6_grad_ds import _relu6_grad_ds_tbe
from .relu_v2 import _relu_v2_tbe
from .relu_grad_v2 import _relu_grad_v2_tbe
from .relu_v2_ds import _relu_v2_ds_tbe
from .relu_grad_v2_ds import _relu_grad_v2_ds_tbe
from .softmax_cross_entropy_with_logits import _softmax_cross_entropy_with_logits_tbe
from .softmax_cross_entropy_with_logits_ds import _softmax_cross_entropy_with_logits_ds_tbe
from .sigmoid_cross_entropy_with_logits import _sigmoid_cross_entropy_with_logits_tbe
from .sigmoid_cross_entropy_with_logits_ds import _sigmoid_cross_entropy_with_logits_ds_tbe
from .sigmoid_cross_entropy_with_logits_grad import _sigmoid_cross_entropy_with_logits_grad_tbe
from .sparse_apply_adadelta import _sparse_apply_adadelta_tbe
from .sigmoid_cross_entropy_with_logits_grad_ds import _sigmoid_cross_entropy_with_logits_grad_ds_tbe
from .trans_data import _trans_data_tbe
from .trans_data_ds import _trans_data_ds_tbe
from .trans_data_rnn import _trans_data_rnn_tbe
from .top_k import _top_k_tbe
from .top_k_ds import _top_k_ds_tbe
from .matmul import _matmul_tbe
from .matmul_ds import _matmul_ds_tbe
from .matmul_v2 import _matmul_v2_tbe
from .sub import _sub_tbe
from .sub_ds import _sub_ds_tbe
from .scatter_nd import _scatter_nd_tbe
from .scatter_nd_ds import _scatter_nd_ds_tbe
from .scatter_nd_d import _scatter_nd_d_tbe
from .scatter_nd_add import _scatter_nd_add_tbe
from .scatter_nd_sub import _scatter_nd_sub_tbe
from .scatter_nd_sub_ds import _scatter_nd_sub_ds_tbe
from .scatter_non_aliasing_add import _scatter_non_aliasing_add_tbe
from .scatter_non_aliasing_add_ds import _scatter_non_aliasing_add_ds_tbe
from .reduce_mean import _reduce_mean_tbe
from .reduce_mean_ds import _reduce_mean_ds_tbe
from .tile import _tile_tbe
from .tile_ds import _tile_ds_tbe
from .atomic_addr_clean import _atomic_addr_clean_tbe
from .dynamic_atomic_addr_clean import _dynamic_atomic_addr_clean_tbe
from .gather_v2 import _gather_v2_tbe
from .gather_v2_ds import _gather_v2_ds_tbe
from .gather_nd import _gather_nd_tbe
from .gather_nd_ds import _gather_nd_ds_tbe
from .bn_training_reduce import _bn_training_reduce_tbe
from .bn_training_reduce_ds import _bn_training_reduce_ds_tbe
from .bn_training_reduce_grad import _bn_training_reduce_grad_tbe
from .bn_training_reduce_grad_ds import _bn_training_reduce_grad_ds_tbe
from .bn_training_update import _bn_training_update_tbe
from .bn_training_update_ds import _bn_training_update_ds_tbe
from .bn_training_update_grad import _bn_training_update_grad_tbe
from .bn_training_update_grad_ds import _bn_training_update_grad_ds_tbe
from .bn_infer import _bn_infer_tbe
from .bn_infer_ds import _bn_infer_ds_tbe
from .bn_infer_grad import _bn_infer_grad_tbe
from .bn_inference import _bn_inference_tbe
from .fused_dbn_dw import _fused_dbn_dw_tbe
from .reciprocal import _reciprocal_tbe
from .reverse_v2_d import _reverse_v2_d_tbe
from .rint import _rint_tbe
from .rint_ds import _rint_ds_tbe
from .strided_slice_d import _strided_slice_d_tbe
from .strided_slice_ds import _strided_slice_ds_tbe
from .strided_slice_grad_d import _strided_slice_grad_d_tbe
from .strided_slice_grad_ds import _strided_slice_grad_ds_tbe
from .split_d import _split_d_tbe
from .split_d_ds import _split_d_ds_tbe
from .exp import _exp_tbe
from .expm1 import _expm1_tbe
from .elu import _elu_tbe
from .elu_ds import _elu_ds_tbe
from .elu_grad import _elu_grad_tbe
from .elu_grad_ds import _elu_grad_ds_tbe
from .div import _div_tbe
from .div_ds import _div_ds_tbe
from .log import _log_tbe
from .xdivy import _xdivy_tbe
from .xdivy_ds import _xdivy_ds_tbe
from .xlogy import _xlogy_tbe
from .xlogy_ds import _xlogy_ds_tbe
from .floor_div import _floor_div_tbe
from .floor_div_ds import _floor_div_ds_tbe
from .zeros_like import _zeros_like_tbe
from .zeros_like_ds import _zeros_like_ds_tbe
from .neg import _neg_tbe
from .npu_clear_float_status import _npu_clear_float_status_tbe
from .npu_get_float_status import _npu_get_float_status_tbe
from .npu_alloc_float_status import _npu_alloc_float_status_tbe
from .one_hot import _one_hot_tbe
from .one_hot_ds import _one_hot_ds_tbe
from .equal import _equal_tbe
from .equal_ds import _equal_ds_tbe
from .lerp import _lerp_tbe
from .less import _less_tbe
from .less_ds import _less_ds_tbe
from .less_equal import _less_equal_tbe
from .less_equal_ds import _less_equal_ds_tbe
from .logical_and import _logical_and_tbe
from .logical_and_ds import _logical_and_ds_tbe
from .logical_not import _logical_not_tbe
from .logical_not_ds import _logical_not_ds_tbe
from .logical_or import _logical_or_tbe
from .logical_or_ds import _logical_or_ds_tbe
from .reduce_max import _reduce_max_tbe
from .reduce_max_ds import _reduce_max_ds_tbe
from .reduce_min import _reduce_min_tbe
from .reduce_min_ds import _reduce_min_ds_tbe
from .reduce_sum import _reduce_sum_tbe
from .reduce_sum_ds import _reduce_sum_ds_tbe
from .reduce_std import _reduce_std_tbe
from .round import _round_tbe
from .tanh import _tanh_tbe
from .tanh_ds import _tanh_ds_tbe
from .tanh_grad import _tanh_grad_tbe
from .tanh_grad_ds import _tanh_grad_ds_tbe
from .softmax import _softmax_tbe
from .softmax_ds import _softmax_ds_tbe
from .softsign import _softsign_tbe
from .softsign_ds import _softsign_ds_tbe
from .softplus import _softplus_tbe
from .softplus_ds import _softplus_ds_tbe
from .softplus_grad import _softplus_grad_tbe
from .softplus_grad_ds import _softplus_grad_ds_tbe
from .softmax_grad_ext import _softmax_grad_ext_tbe
from .square import _square_tbe
from .square_ds import _square_ds_tbe
from .squared_difference import _squared_difference_tbe
from .squared_difference_ds import _squared_difference_ds_tbe
from .sqrt import _sqrt_tbe
from .sqrt_ds import _sqrt_ds_tbe
from .sparse_apply_ftrl_d import _sparse_apply_ftrl_d
from .sparse_apply_ftrl_d_ds import _sparse_apply_ftrl_d_ds
from .sparse_apply_proximal_adagrad import _sparse_apply_proximal_adagrad
from .sparse_apply_proximal_adagrad_ds import _sparse_apply_proximal_adagrad_ds
from .apply_proximal_adagrad import _apply_proximal_adagrad
from .apply_proximal_adagrad_ds import _apply_proximal_adagrad_ds_tbe
from .transpose_nod import _transpose_nod_tbe
from .transpose_d import _transpose_d_tbe
from .transpose import _transpose_tbe
from .truncate_div import _truncate_div_tbe
from .truncate_div_ds import _truncate_div_ds_tbe
from .truncate_mod import _truncate_mod_tbe
from .truncate_mod_ds import _truncate_mod_ds_tbe
from .unsorted_segment_sum import _unsorted_segment_sum_tbe
from .unsorted_segment_sum_ds import _unsorted_segment_sum_ds_tbe
from .unsorted_segment_prod import _unsorted_segment_prod_tbe
from .unsorted_segment_prod_ds import _unsorted_segment_prod_ds_tbe
from .logsoftmax_grad import _logsoftmax_grad_tbe
from .logsoftmax_grad_ds import _logsoftmax_grad_ds_tbe
from .logsoftmax import _logsoftmax_tbe
from .logsoftmax_ds import _logsoftmax_ds_tbe
from .select import _select_tbe
from .select_ds import _select_ds_tbe
from .pow import _pow_tbe
from .pow_ds import _pow_ds_tbe
from .maximum import _maximum_tbe
from .maximum_ds import _maximum_ds_tbe
from .maximum_grad_ds import _maximum_grad_ds_tbe
from .minimum import _minimum_tbe
from .minimum_ds import _minimum_ds_tbe
from .minimum_grad import _minimum_grad_tbe
from .maximum_grad import _maximum_grad_tbe
from .minimum_grad_ds import _minimum_grad_ds_tbe
from .concat import _concat_tbe
from .concat_ds import _concat_ds_tbe
from .slice import _slice_tbe
from .sign import _sign_tbe
from .sign_ds import _sign_ds_tbe
from .greater import _greater_tbe
from .clip_by_norm_no_div_sum import _clip_by_norm_no_div_sum_tbe
from .clip_by_norm_no_div_sum_ds import _clip_by_norm_no_div_sum_ds_tbe
from .clip_by_value import _clip_by_value_tbe
from .clip_by_value_ds import _clip_by_value_ds_tbe
from .layer_norm_beta_gamma_backprop import _layer_norm_beta_gamma_backprop_tbe
from .layer_norm_beta_gamma_backprop_ds import _layer_norm_beta_gamma_backprop_ds_tbe
from .layer_norm_beta_gamma_backprop_v2 import _layer_norm_beta_gamma_backprop_v2_tbe
from .layer_norm_beta_gamma_backprop_v2_ds import _layer_norm_beta_gamma_backprop_v2_ds_tbe
from .layer_norm import _layer_norm_tbe
from .layer_norm_ds import _layer_norm_ds_tbe
from .layer_norm_grad import _layer_norm_grad_tbe
from .layer_norm_x_backprop_ds import _layer_norm_x_backprop_ds_tbe
from .layer_norm_x_backprop_v2 import _layer_norm_x_backprop_v2_tbe
from .layer_norm_x_backprop_v2_ds import _layer_norm_x_backprop_v2_ds_tbe
from .kl_div_loss import _kl_div_loss_tbe
from .kl_div_loss_ds import _kl_div_loss_ds_tbe
from .kl_div_loss_grad import _kl_div_loss_grad_tbe
from .l2_loss import _l2_loss_tbe
from .l2_loss_ds import _l2_loss_ds_tbe
from .l2_normalize import _l2_normalize_tbe
from .l2_normalize_grad import _l2_normalize_grad_tbe
from .square_sum_v1 import _square_sum_v1_tbe
from .square_sum_v1_ds import _square_sum_v1_ds_tbe
from .square_sum_v2 import _square_sum_v2_tbe
from .confusion_transpose_d import _confusion_transpose_d_tbe
from .confusion_softmax_grad import _confusion_softmax_grad_tbe
from .lamb_update_with_lr_v2 import _lamb_update_with_lr_v2_tbe
from .lamb_next_mv import _lamb_next_mv_tbe
from .lamb_next_mv_with_decay import _lamb_next_mv_with_decay_tbe
from .lamb_update_with_lr import _lamb_update_with_lr_tbe
from .rsqrt import _rsqrt_tbe
from .rsqrt_ds import _rsqrt_ds_tbe
from .sigmoid import _sigmoid_tbe
from .sigmoid_ds import _sigmoid_ds_tbe
from .sigmoid_grad import _sigmoid_grad_tbe
from .sigmoid_grad_ds import _sigmoid_grad_ds_tbe
from .resize_nearest_neighbor import _resize_nearest_neighbor_tbe
from .resize_nearest_neighbor_ds import _resize_nearest_neighbor_ds_tbe
from .resize_nearest_neighbor_grad import _resize_nearest_neighbor_grad_tbe
from .resize_nearest_neighbor_grad_ds import _resize_nearest_neighbor_grad_ds_tbe
from .pad_d import _pad_d_tbe
from .pad_d_ds import _pad_d_ds_tbe
from .arg_max_with_value import _arg_max_with_value_tbe
from .arg_max_with_value_ds import _arg_max_with_value_ds_tbe
from .arg_min_with_value import _arg_min_with_value_tbe
from .smooth_l1_loss import _smooth_l1_loss_tbe
from .smooth_l1_loss_ds import _smooth_l1_loss_ds_tbe
from .smooth_l1_loss_grad import _smooth_l1_loss_grad_tbe
from .smooth_l1_loss_grad_ds import _smooth_l1_loss_grad_ds_tbe
from .soft_margin_loss import _soft_margin_loss_tbe
from .soft_margin_loss_grad import _soft_margin_loss_grad_tbe
from .fused_mul_add import _fused_mul_add_tbe
from .fused_mul_add_n import _fused_mul_add_n_tbe
from .fused_mul_apply_momentum import _fused_mul_apply_momentum_tbe
from .fill import _fill_op_tbe
from .fill_ds import _fill_ds_op_tbe
from .erf import _erf_op_tbe
from .erf_ds import _erf_ds_op_tbe
from .erfc import _erfc_op_tbe
from .erfc_ds import _erfc_ds_op_tbe
from .depthwise_conv2d import _depthwise_conv2d_tbe
from .depthwise_conv2d_backprop_filter import _depthwise_conv2d_backprop_filter_tbe
from .depthwise_conv2d_backprop_input import _depthwise_conv2d_backprop_input_tbe
from .greater_equal import _greater_equal_tbe
from .greater_equal_ds import _greater_equal_ds_tbe
from .not_equal import _not_equal_tbe
from .floor_mod import _floor_mod_tbe
from .floor_mod_ds import _floor_mod_ds_tbe
from .scatter_nd_update import _scatter_nd_update_tbe
from .scatter_nd_update_ds import _scatter_nd_update_ds_tbe
from .avg_pool import _avg_pool_tbe
from .avg_pool_ds import _avg_pool_ds_tbe
from .avg_pool_grad import _avg_pool_grad_tbe
from .avg_pool_grad_vm import _avg_pool_grad_vm_tbe
from .avg_pool_3d import _avg_pool_3d_tbe
from .avg_pool_3d_grad import _avg_pool_3d_grad_tbe
from .ones_like import _ones_like_tbe
from .ones_like_ds import _ones_like_ds_tbe
from .batch_to_space import _batch_to_space_tbe
from .space_to_batch import _space_to_batch_tbe
from .depth_to_space_ds import _depth_to_space_ds_tbe
from .space_to_depth import _space_to_depth_tbe
from .extract_image_patches import _extract_image_patches_tbe
from .sort import _sort_tbe
from .sort_ds import _sort_ds_tbe
from .floor import _floor_tbe
from .floor_ds import _floor_ds_tbe
from .ceil import _ceil_tbe
from .ceil_ds import _ceil_ds_tbe
from .log1p import _log1p_tbe
from .log1p_ds import _log1p_ds_tbe
from .resize_bilinear import _resize_bilinear_tbe
from .resize_bilinear_v2 import _resize_bilinear_v2_tbe
from .resize_bilinear_grad import _resize_bilinear_grad_tbe
from .flatten import _flatten_tbe
from .roi_align import _roi_align_tbe
from .roi_align_ds import _roi_align_ds_tbe
from .roi_align_grad import _roi_align_grad_tbe
from .roi_align_grad_ds import _roi_align_grad_ds_tbe
from .bounding_box_decode import _bounding_box_decode_tbe
from .bounding_box_decode_ds import _bounding_box_decode_ds_tbe
from .bounding_box_encode import _bounding_box_encode_tbe
from .check_valid import _check_valid_tbe
from .check_valid_ds import _check_valid_ds_tbe
from .iou import _iou_tbe
from .iou_ds import _iou_ds_tbe
from .arg_max import _arg_max_tbe
from .nms_with_mask import _nms_with_mask_tbe
from .sgd import _sgd_tbe
from .lars_update import _lars_update_tbe
from .lars_update_ds import _lars_update_ds_tbe
from .arg_min import _arg_min_tbe
from .arg_min_v2_ds import _arg_min_v2_ds_tbe
from .bn_training_update_v2 import _bn_training_update_v2_tbe
from .bn_training_update_v3 import _bn_training_update_v3_tbe
from .square_sum_all import _square_sum_all_tbe
from .square_sum_all_ds import _square_sum_all_ds_tbe
from .pack import _pack_tbe
from .pack_ds import _pack_ds_tbe
from .unpack import _unpack_tbe
from .unpack_ds import _unpack_ds_tbe
from .scatter_update import _scatter_update_tbe
from .prelu import _prelu_tbe
from .prelu_ds import _prelu_ds_tbe
from .prelu_grad import _prelu_grad_tbe
from .binary_cross_entropy_ds import _binary_cross_entropy_ds_tbe
from .binary_cross_entropy import _binary_cross_entropy_tbe
from .binary_cross_entropy_grad import _binary_cross_entropy_grad_tbe
from .binary_cross_entropy_grad_ds import _binary_cross_entropy_grad_ds_tbe
from .bce_with_logits_loss import _bce_with_logits_loss_op_tbe
from .sin import _sin_tbe
from .sin_ds import _sin_ds_tbe
from .cos_ds import _cos_ds_tbe
from .cos import _cos_tbe
from .tan import _tan_tbe
from .tan_ds import _tan_ds_tbe
from .cum_sum import _cum_sum_tbe
from .cum_sum_ds import _cum_sum_ds_tbe
from .apply_rms_prop import _apply_rms_prop_tbe
from .cumprod import _cumprop_tbe
from .reduce_prod import _reduce_prod_tbe
from .reduce_prod_ds import _reduce_prod_ds_tbe
from .reciprocal_grad import _reciprocal_grad_tbe
from .reciprocal_grad_ds import _reciprocal_grad_ds_tbe
from .sqrt_grad import _sqrt_grad_tbe
from .sqrt_grad_ds import _sqrt_grad_ds_tbe
from .rsqrt_grad import _rsqrt_grad_tbe
from .rsqrt_grad_ds import _rsqrt_grad_ds_tbe
from .scatter_add import _scatter_add_tbe
from .atan2 import _atan2_tbe
from .atan2_ds import _atan2_ds_tbe
from .bessel_i0e import _bessel_i0e_tbe
from .bessel_i0e_ds import _bessel_i0e_ds_tbe
from .bessel_i1e import _bessel_i1e_tbe
from .bessel_i1e_ds import _bessel_i1e_ds_tbe
from .batch_to_space_nd import _batch_to_space_nd_tbe
from .batch_to_space_nd_ds import _batch_to_space_nd_ds_tbe
from .batch_to_space_nd_v2 import _batch_to_space_nd_v2_tbe
from .space_to_batch_nd import _space_to_batch_nd_tbe
from .bitwise_and import _bitwise_and_tbe
from .bitwise_and_ds import _bitwise_and_ds_tbe
from .bitwise_or import _bitwise_or_tbe
from .bitwise_or_ds import _bitwise_or_ds_tbe
from .bitwise_xor import _bitwise_xor_tbe
from .bitwise_xor_ds import _bitwise_xor_ds_tbe
from .reduce_all import _reduce_all_tbe
from .reduce_all_ds import _reduce_all_ds_tbe
from .reduce_any import _reduce_any_tbe
from .reduce_any_ds import _reduce_any_ds_tbe
from .sparse_apply_adagrad import _sparse_apply_adagrad_tbe
from .sparse_apply_adagrad_ds import _sparse_apply_adagrad_ds
from .sparse_apply_adagrad_v2_ds import _sparse_apply_adagrad_v2_ds
from .unsorted_segment_min import _unsorted_segment_min_tbe
from .unsorted_segment_min_ds import _unsorted_segment_min_ds_tbe
from .unsorted_segment_max import _unsorted_segment_max_tbe
from .unsorted_segment_max_ds import _unsorted_segment_max_ds_tbe
from .asin import _asin_tbe
from .asin_ds import _asin_ds_tbe
from .asin_grad import _asin_grad_tbe
from .asin_grad_ds import _asin_grad_ds_tbe
from .asinh import _asinh_tbe
from .asinh_ds import _asinh_ds_tbe
from .asinh_grad import _asinh_grad_tbe
from .asinh_grad_ds import _asinh_grad_ds_tbe
from .div_no_nan import _div_no_nan_tbe
from .div_no_nan_ds import _div_no_nan_ds_tbe
from .atan import _atan_tbe
from .atan_ds import _atan_ds_tbe
from .atan_grad import _atan_grad_tbe
from .atan_grad_ds import _atan_grad_ds_tbe
from .atanh import _atanh_tbe
from .atanh_ds import _atanh_ds_tbe
from .cosh import _cosh_tbe
from .cosh_ds import _cosh_ds_tbe
from .sinh import _sinh_tbe
from .sinh_ds import _sinh_ds_tbe
from .inv import _inv_tbe
from .inv_ds import _inv_ds_tbe
from .inv_grad import _inv_grad_tbe
from .inv_grad_ds import _inv_grad_ds_tbe
from .invert import _invert_tbe
from .invert_ds import _invert_ds_tbe
from .basic_lstm_cell import _basic_lstm_cell_tbe
from .basic_lstm_cell_c_state_grad_v2 import _basic_lstm_cell_c_state_grad_tbe_v2
from .basic_lstm_cell_c_state_grad import _basic_lstm_cell_c_state_grad_tbe
from .basic_lstm_cell_weight_grad import _basic_lstm_cell_weight_grad_tbe
from .basic_lstm_cell_input_grad import _basic_lstm_cell_input_grad_tbe
from .dynamic_rnn import _dynamic_rnn_tbe
from .dynamic_gru_v2 import _dynamic_gru_v2_tbe
from .gru_v2_hidden_grad import _gru_v2_hidden_grad_tbe
from .gru_v2_hidden_grad_cell import _gru_v2_hidden_grad_cell_tbe
from .lstm_input_grad import _lstm_input_grad_tbe
from .confusion_matrix import _confusion_matrix_tbe
from .broadcast_to import _broadcast_to_tbe
from .broadcast_to_ds import _broadcast_to_ds_tbe
from .strided_read import _strided_read_tbe
from .strided_write import _strided_write_tbe
from .range import _range_tbe
from .fused_mul_add_n_l2loss import _fused_mul_add_n_l2loss_tbe
from .fused_mul_apply_momentum_extern import _fused_mul_apply_momentum_extern_tbe
from .lamb_next_right import _lamb_next_right_tbe
from .sparse_gather_v2 import _sparse_gather_v2_tbe
from .sparse_gather_v2_ds import _sparse_gather_v2_ds_tbe
from .sparse_apply_r_m_s_prop import _sparse_apply_r_m_s_prop_tbe
from .sparse_apply_r_m_s_prop_ds import _sparse_apply_r_m_s_prop_ds_tbe
from .data_format_dim_map import _data_format_dim_map_tbe
from .data_format_dim_map_ds import _data_format_dim_map_ds_tbe
from .histogram_fixed_width import _histogram_fixed_width_tbe
from .tensor_scatter_update import _tensor_scatter_update_tbe
from .inplace_update import _inplace_update_tbe
from .inplace_update_ds import _inplace_update_v2_ds_tbe
from .split_v import _split_v_tbe
from .in_top_k import _in_top_k_tbe
from .diag import _diag_tbe
from .diag_part import _diag_part_tbe
from .matrix_diag import _matrix_diag_tbe
from .matrix_diag_part import _matrix_diag_part_tbe
from .matrix_set_diag import _matrix_set_diag_tbe
from .lrn import _lrn_tbe
from .lrn_grad import _lrn_grad_tbe
from .scatter_max import _scatter_max_tbe
from .scatter_min import _scatter_min_tbe
from .scatter_sub import _scatter_sub_tbe
from .scatter_sub_ds import _scatter_sub_ds_tbe
from .scatter_mul import _scatter_mul_tbe
from .scatter_div import _scatter_div_tbe
from .mod import _mod_tbe
from .mod_ds import _mod_ds_tbe
from .max_pool_grad_grad import _max_pool_grad_grad_tbe
from .max_pool_grad_grad_with_argmax import _max_pool_grad_grad_with_argmax_tbe
from .tensor_move import _tensor_move_tbe
from .tensor_move_ds import _tensor_move_ds_tbe
from .population_count import _population_count_tbe
from .parallel_concat import _parallel_concat_tbe
from .adam_apply_one_assign import _adam_apply_one_assign_tbe
from .adam_apply_one_with_decay_assign import _adam_apply_one_with_decay_assign_tbe
from .ifmr import _ifmr_tbe
from .ifmr_ds import _ifmr_ds_tbe
from .acts_ulq import _acts_ulq_tbe
from .acts_ulq_input_grad import _acts_ulq_input_grad_tbe
from .act_ulq_clamp_min_grad import _act_ulq_clamp_min_grad_tbe
from .act_ulq_clamp_max_grad import _act_ulq_clamp_max_grad_tbe
from .wts_arq import _wts_arq_tbe
from .fake_quant_with_min_max_vars import _fake_quant_with_min_max_vars_tbe
from .fake_quant_with_min_max_vars_gradient import _fake_quant_with_min_max_vars_gradient_tbe
from .fake_quant_with_min_max_vars_per_channel import _fake_quant_with_min_max_vars_per_channel_tbe
from .fake_quant_with_min_max_vars_per_channel_gradient import _fake_quant_with_min_max_vars_per_channel_gradient_tbe
from .conv3d import _conv3d_tbe
from .conv3d_backprop_input import _conv3d_backprop_input_tbe
from .conv3d_backprop_filter import _conv3d_backprop_filter_tbe
from .conv3d_transpose import _conv3d_transpose_tbe
from .conv3d_transpose_ds import _conv3d_transpose_ds_tbe
from .lamb_apply_optimizer_assign import _lamb_apply_optimizer_assign_tbe
from .lamb_apply_weight_assign import _lamb_apply_weight_assign_tbe
from .nll_loss import _nll_loss_tbe
from .nll_loss_grad import _nll_loss_grad_tbe
from .masked_fill import _masked_fill_tbe
from .masked_fill_ds import _masked_fill_ds_tbe
from .mish import _mish_tbe
from .mul_no_nan import _mul_no_nan_tbe
from .mul_no_nan_ds import _mul_no_nan_ds_tbe
from .selu import _selu_tbe
from .selu_ds import _selu_ds_tbe
from .centralization import _centralization_tbe
from .exp_ds import _exp_ds_tbe
from .log_ds import _log_ds_tbe
from .cdist import _cdist_tbe
from .cdist_grad import _cdist_grad_tbe
from .pdist import _pdist_tbe
from .neg_ds import _neg_ds_tbe
from .not_equal_ds import _not_ds_equal_tbe
from .reciprocal_ds import _reciprocal_ds_tbe
from .lp_norm import _lp_norm_tbe
from .lp_norm_ds import _lp_norm_ds_tbe
from .ctc_loss_v2 import _ctc_loss_v2_tbe
from .ctc_loss_v2_grad import _ctc_loss_v2_grad_tbe
from .roll import _roll_tbe
from .soft_shrink import _soft_shrink_tbe
from .erfinv import _erfinv_tbe
from .soft_shrink_grad import _soft_shrink_grad_tbe
from .index_add import _index_add_tbe
from .hsigmoid_grad import _hsigmoid_grad_tbe
from .hsigmoid import _hsigmoid_tbe
from .hshrink import _hshrink_tbe
from .hshrink_grad import _hshrink_grad_tbe
from .new_im2col import _new_im2col_tbe
from .trunc import _trunc_tbe
from .extract_volume_patches import _extract_volume_patches_tbe
from .multilabel_margin_loss import _multilabel_margin_loss_tbe
from .round_ds import _round_ds_tbe
from .is_close import _is_close_tbe
from .apply_adam_with_amsgrad import _apply_adam_with_amsgrad_tbe
from .apply_adam_with_amsgrad_ds import _apply_adam_with_amsgrad_ds_tbe
from .expm1_ds import _expm1_ds_tbe
from .deformable_offsets import _deformable_offsets_tbe
from .deformable_offsets_grad import _deformable_offsets_grad_tbe
from .parallel_resize_bilinear import _parallel_resize_bilinear_op_info_tbe
from .parallel_resize_bilinear_grad import _parallel_resize_bilinear_grad_op_info_tbe
from .p_s_r_o_i_pooling import _p_s_r_o_i_pooling_tbe
from .p_s_r_o_i_pooling_grad import _p_s_r_o_i_pooling_grad_tbe
from .renorm import _renorm_tbe
from .adam_apply_one_ds import _adam_apply_one_ds_tbe
from .adam_apply_one_with_decay_ds import _adam_apply_one_with_decay_ds_tbe
from .adaptive_max_pool2d import _adaptive_max_pool2d_tbe
from .pooling import _pooling_tbe
from .hard_swish import _hard_swish_tbe
from .hard_swish_grad import _hard_swish_grad_tbe
from .hard_swish_ds import _hard_swish_ds_tbe
from .hard_swish_grad_ds import _hard_swish_grad_ds_tbe
from .arg_min_with_value_ds import _arg_min_with_value_ds_tbe
from .im2col import _im2col_tbe
